package com.xvyy.tingshu.user.service.impl;

import java.util.Date;

import com.alibaba.fastjson.JSONObject;
import com.xvyy.tingshu.common.constant.SystemConstant;
import com.xvyy.tingshu.common.rabbit.constant.MqConst;
import com.xvyy.tingshu.common.rabbit.service.RabbitService;
import com.xvyy.tingshu.common.util.AuthContextHolder;
import com.xvyy.tingshu.common.util.MongoUtil;
import com.xvyy.tingshu.model.user.UserListenProcess;
import com.xvyy.tingshu.user.service.UserListenProcessService;
import com.xvyy.tingshu.vo.album.TrackStatMqVo;
import com.xvyy.tingshu.vo.user.UserListenProcessVo;
import lombok.extern.slf4j.Slf4j;
import org.bson.types.ObjectId;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.TimeUnit;


@Slf4j
@Service
@SuppressWarnings({"unchecked", "rawtypes"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    @Autowired
    private RabbitService rabbitService;

    @Override
    public BigDecimal getTrackBreakSecond(Long trackId) {

        Long userId = AuthContextHolder.getUserId();
        // 1. 构建条件对象
        Criteria criteria = Criteria.where("userId").is(userId).and("trackId").is(trackId);
        // 2. 构建一个查询对象
        Query query = new Query(criteria);
        // 3. 查询
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (userListenProcess == null) {
            // 没播放
            return new BigDecimal("0.00");
        }
        // 播放
        return userListenProcess.getBreakSecond();  // 存放到mongo中的播放秒数
    }

    @Override
    public void updateListenProcess(UserListenProcessVo userListenProcessVo) {

        // 1. 获取当前用户id
        Long userId = AuthContextHolder.getUserId();
        // 2. 构建条件对象
        Criteria criteria = Criteria.where("userId").is(userId).and("trackId").is(userListenProcessVo.getTrackId());
        // 3. 构建一个查询对象
        Query query = new Query(criteria);
        // 4. 查询
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (userListenProcess == null) {
            // 第一次播放，创建记录
            userListenProcess = new UserListenProcess();// 创建对象
            userListenProcess.setId(ObjectId.get().toString());// 唯一id
            userListenProcess.setUserId(userId);// 当前用户id
            userListenProcess.setAlbumId(userListenProcessVo.getAlbumId());// 播放的专辑id
            userListenProcess.setTrackId(userListenProcessVo.getTrackId());// 播放的曲目id
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());// 播放到哪一秒
            userListenProcess.setIsShow(1);// 默认显示
            userListenProcess.setCreateTime(new Date());// 第一次播放，创建时间和更新时间是一样的
            userListenProcess.setUpdateTime(new Date());// 第一次播放，创建时间和更新时间是一样的
            // 保存到MongoDB中
            mongoTemplate.save(userListenProcess, collectionName);
            log.info("创建播放记录到MongoDB中成功");
        } else {
            // 播放过，更新记录
            userListenProcess.setUpdateTime(new Date());// 更新时间
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());// 更新播放到哪一秒
            // 更新到MongoDB中
            mongoTemplate.save(userListenProcess, collectionName);
            log.info("更新播放记录到MongoDB中成功");
        }

        // 5. 修改MySQL中声音/专辑的播放次数(异步处理)
        // 5.1 构建消息对象
        // 这里采用redis中的bitmap来实现
        String bitmapKey = new DateTime().toString("yyyyMMdd") + ":" + userListenProcessVo.getTrackId();

        Boolean aBoolean = stringRedisTemplate.opsForValue().setBit(bitmapKey, userId, true);
        if (!aBoolean) {
            // 当天第一次播放
            stringRedisTemplate.expire(bitmapKey, 1, TimeUnit.DAYS);
            TrackStatMqVo trackStatMqVo = new TrackStatMqVo();
            trackStatMqVo.setBusinessNo(UUID.randomUUID().toString().replaceAll("-", ""));// 唯一标识 业务流水号
            trackStatMqVo.setAlbumId(userListenProcessVo.getAlbumId());
            trackStatMqVo.setTrackId(userListenProcessVo.getTrackId());
            trackStatMqVo.setStatType(SystemConstant.ALBUM_STAT_PLAY);
            trackStatMqVo.setCount(1);
            rabbitService.sendMessage(
                    MqConst.EXCHANGE_TRACK, MqConst.ROUTING_TRACK_STAT_UPDATE, JSONObject.toJSONString(trackStatMqVo));
        }

    }

    @Override
    public Map<String, Object> getLatelyTrack() {
        Long userId = AuthContextHolder.getUserId();
        // 1. 构建条件对象
        Criteria criteria = Criteria.where("userId").is(userId);
        // 2. 构建一个查询对象
        Query query = new Query(criteria);
        // 3. 根据id找到修改时间最新记录
        Sort sort = Sort.by(Sort.Direction.DESC, "updateTime");
        query.with(sort);
        // 4. 查询
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        Map<String, Object> resultMap = new HashMap<>();
        if (userListenProcess == null) {
            // 没播放过，返回空对象
            return resultMap;
        }
        resultMap.put("albumId", userListenProcess.getAlbumId());// 播放的专辑id
        resultMap.put("trackId", userListenProcess.getTrackId());// 播放的曲目id
        return resultMap;
    }
}
